tensor([[0.5300, 0.5300, 0.5300, 0.5300, 0.5300],
        [0.4700, 0.4700, 0.4700, 0.4700, 0.4700]], grad_fn=<RoundBackward1>)